#ifndef _RAY_UTIL_H
#define _RAY_UTIL_H

#include <torch/torch.h>
#include <cuda.h>
#include <cuda_runtime.h>
#include <ATen/ATen.h>

#include <cstdio>
#include <stdint.h>

extern "C" __declspec(dllexport) void rays_coords(const at::Tensor* rays_o, const at::Tensor* rays_d, const at::Tensor* aabb,
	const uint32_t num_steps, const float min_near, const uint32_t N, const bool perturb,
	at::Tensor* nears, at::Tensor* fars, at::Tensor* xyzs, at::Tensor* z_vals, at::Tensor* deltas);

#endif